import pandas as pd

# 分拆X_train, y_train, X_test, y_test
train = pd.read_pickle("pkl/train_fe")
print(train.shape)

y_train = train['click']
y_train.to_pickle("pkl/y_train")
print("saved y train")
X_train = train.drop(["click"], axis=1)
X_train.to_pickle("pkl/X_train")

test = pd.read_pickle("pkl/test_fe")
print(test.shape)
test = test.drop(["id"], axis=1)
test.to_pickle("pkl/X_test")
